package com.leicx.guli.order.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.boot.SpringBootConfiguration;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Primary;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * rabbitMq 序列化器
 *
 * @author daxiong
 * @date 2022/8/8 下午1:55
 */
@SpringBootConfiguration
@Slf4j
public class MyRabbitMqConfig {

    private RabbitTemplate rabbitTemplate;

    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        this.rabbitTemplate = rabbitTemplate;
        rabbitTemplate.setMessageConverter(messageConverter());
        initRabbitTemplate();
        return rabbitTemplate;
    }

    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1、broker收到消息就回调：
     * 1.1 spring.rabbitmq.publisher-confirm=true
     * 1.2 设置确认回调ConfirmCallback
     * <p>
     * 2、消息正确抵达队列进行回调
     * 2.1 spring.rabbitmq.publisher-returns=true
     * spring.rabbitmq.template.mandatory=true
     * 2.2 设置确认回调ReturnCallback
     * <p>
     * 3、消费端确认回调（保证每个消息被正常消费，此时才可以broker删除这个消息）
     * 3.1 默认是自动确认的，只要收到消息，服务端就会移除这个消息
     * 存在的问题：收到很多消息，只有一个消息外理成功了，如果此时发生了宕机，就会产生消息丢失。因为channel已经获取到了所有的消息
     * 3.2 手动确认模式，只要没有ack，消息就一直是unack状态，即使发生了宕机
     */
//    @PostConstruct
    public void initRabbitTemplate() {
//        rabbitTemplate.setMessageConverter(new Jackson2JsonMessageConverter());
        // broker收到消息就回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * @param correlationData   当前消息的唯一ID
             * @param ack   消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                log.info("接收到confirm消息：correlationData[{}], ack[{}], cause[{}]", correlationData, ack, cause);
            }
        });

        // 发送端消息抵达队列的确认
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message   消息的详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange  哪个交换机
             * @param routingKey 路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                log.info("fail message[{}], replyCode[{}], replyText[{}], exchange[{}], routingKey[{}]",
                        message, replyCode, replyText, exchange, routingKey);
            }
        });

    }

}
